package com.cpd.blog.common.handler;

import com.cpd.blog.common.constant.StatusCode;
import com.cpd.blog.dto.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * 全局异常处理
 * 对加了 @Controller注解的方法进行拦截处理 （AOP的实现）
 *
 * @author Copydown
 * @date 2021/8/16 16:57
 */
@Slf4j
@ControllerAdvice
public class AllExceptionHandler {

    /**
     * 全局异常处理
     */
    @ExceptionHandler(Exception.class)
    @ResponseBody
    public Result doException(Exception e) {

        Result failResult = Result.fail(StatusCode.SYSTEM_ERROR.getCode(), "");

        if (e instanceof MethodArgumentNotValidException) {
            BindingResult bindResult = ((MethodArgumentNotValidException) e).getBindingResult();
            failResult.setCode(StatusCode.VALIDATED_ERROR.getCode());
            failResult.setMsg(bindResult.getAllErrors().get(0).getDefaultMessage());
        } else {
            e.printStackTrace();
        }
        return failResult;
    }
}
